package com.shazhu.common.utils.nongjt;

import java.util.Random;

public class ValidateCodeUtils {

        /**
         * 随机⽣成验证码
         *
         * @param length ⻓度为4位或者6位
         * @return
         */
        public static String generateValidateCode(int length) {
            Integer code = null;
            if (length == 4) {
                // ⽣成随机数，最⼤为9999
                code = new Random().nextInt(9999);
                if (code < 1000) {
                    // 保证随机数为4位数字
                    code = code + 1000;
                }
            } else if (length == 6) {
                // ⽣成随机数，最⼤为999999
                code = new Random().nextInt(999999);
                if (code < 100000) {
                    // 保证随机数为6位数字
                    code = code + 100000;
                }
            } else {
                throw new RuntimeException("只能⽣成4位或6位数字验证码");
            }
            return String.valueOf(code);
        }
        /**
         * 随机⽣成指定⻓度字符串验证码 带字⺟
         *
         * @param length ⻓度
         * @return
         */
        public static String generateValidateCodeString(int length) {
            Random rdm = new Random();
            String hash1 = Integer.toHexString(rdm.nextInt());
            String capstr = hash1.substring(0, length);
            return capstr;
        }
    }

